package ztcp

import (
	"errors"
	"go.uber.org/zap"
	"sync"
	"venus/util/zlog"
)

// TcpConnManager 连接管理模块
type TcpConnManager struct {
	connections map[uint32]*TcpConnection //管理的连接信息
	connLock    sync.RWMutex              //读写连接的读写锁
}

// NewTcpConnManager 创建一个链接管理
func NewTcpConnManager() *TcpConnManager {
	return &TcpConnManager{
		connections: make(map[uint32]*TcpConnection),
	}
}

// Add 添加链接
func (connMgr *TcpConnManager) Add(conn *TcpConnection) {
	//保护共享资源Map 加写锁
	connMgr.connLock.Lock()
	defer connMgr.connLock.Unlock()

	//将conn连接添加到ConnMananger中
	connMgr.connections[conn.GetConnID()] = conn

	//zlog.Debug("connection add to ConnManager successfully: conn num = ", connMgr.Len())
}

// Remove 删除连接
func (connMgr *TcpConnManager) Remove(conn *TcpConnection) {
	//保护共享资源Map 加写锁
	connMgr.connLock.Lock()
	defer connMgr.connLock.Unlock()

	//删除连接信息
	delete(connMgr.connections, conn.GetConnID())

	//zlog.Debug("connection Remove ConnID=", conn.GetConnID(), " successfully: conn num = ", connMgr.Len())
}

// Get 利用ConnID获取链接
func (connMgr *TcpConnManager) Get(connID uint32) (*TcpConnection, error) {
	//保护共享资源Map 加读锁
	connMgr.connLock.RLock()
	defer connMgr.connLock.RUnlock()

	if conn, ok := connMgr.connections[connID]; ok {
		return conn, nil
	} else {
		return nil, errors.New("connection not found")
	}
}

// Len 获取当前连接
func (connMgr *TcpConnManager) Len() int {
	return len(connMgr.connections)
}

// ClearConn 清除并停止所有连接
func (connMgr *TcpConnManager) ClearConn() {
	//保护共享资源Map 加写锁
	connMgr.connLock.Lock()
	defer connMgr.connLock.Unlock()

	//停止并删除全部的连接信息
	for connID, conn := range connMgr.connections {
		//停止
		conn.Stop()
		//删除
		delete(connMgr.connections, connID)
	}

	zlog.Debug("Clear All Connections successfully", zap.Int("conn num", connMgr.Len()))
}
